pytorch环境配置 一键官网配置+离线配置(anaconda+duda+cudnn+pytorch) |
您所在的位置:网站首页 › anaconda 离线创建环境 › pytorch环境配置 一键官网配置+离线配置(anaconda+duda+cudnn+pytorch) |
pytorch环境配置(anaconda+duda+cudnn+pytorch) 目录 一:pytorch官网一键法 二:自行离线配置法 一:pytorch官网一键法1、准备工作: 下载anaconda,后续会用到其中的jupyter notebook和anaconda prompt。 2、创建环境: 原因:为了避免冲突及容纳多版本pytorch,所以在base的基础上创建需要的python版本环境。 操作: (1)打开anaconda prompt,在base环境中创建 conda create -n environment_name(自定义) python=3.7。等待创建完毕。 (2)进入创建的环境 输入conda activate environment_name 3、安装pytorch: pytorch安装非常的人性化,进入pytorch官网,点击install,选择所需要的选项: 对于每一项的选择在这里做一下详细介绍: ① Pytorch Build:建议选Stable稳定版 ② Your OS:根据你自己系统选择,这里我选Windows ③ Package:因为我们已经安装Anaconda,所以在Anaconda Prompt环境下我们可以直接使用Conda方式进行安装,所以选Conda ④ Language:Python ⑤ CUDA:关于GPU的CUDA版本问题如何选择: 第一种方式:打开cmd,输入nvidia-smi,可以看到CUDA的版本信息(没有英伟达显卡,或者显卡不支持CUDA或者就想用CPU运行的话直接选None就行)若此方法不行,可选择以下方法。 第二种方法:进入自己电脑的NVIDIA设置,点击帮助-系统信息,查看自己的版本信息。再与下表对应选择即可。如版本为451.67,则可选择10.1,10.2,11.0等。 若以上方法实在太慢,可以参考以下方法。 二:自行离线配置法1、准备工作: 下载anaconda,后续会用到其中的jupyter notebook和anaconda prompt。 2、创建环境: 原因:为了避免冲突及容纳多版本pytorch,所以在base的基础上创建需要的python版本环境。 操作: (1)打开anaconda prompt,在base环境中创建 conda create -n environment_name(自定义) python=3.7。等待创建完毕。 (2)进入创建的环境 输入conda activate environment_name 3、下载安装cuda python虚拟环境创建、cuda环境配置和pytorch1.7.1安装_哔哩哔哩_bilibili 4、下载安装cudnn 见以上链接 5、下载安装pytorch 见以上链接 也可以参考这个链接:安装PyTorch GPU/CPU 两种方法,各种版本,稳稳的!_哔哩哔哩_bilibili 注: 1、验证是否安装成功: conda list指令查找一下有没有pytorch 输入 python,进入 python 环境,输入 import pytorch,如果没有报错,说明安装成功。再输入 torch.cuda.is_available() 检查 GPU 是否可以被 pytorch 使用。 2、换源安装: conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/ # 以上两条是Anaconda官方库的镜像 conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/ # 以上是Anaconda第三方库 Conda Forge的镜像 # for linux conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/ # for legacy win-64 conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/peterjc123/ 以上两条是Pytorch的Anaconda第三方镜像 conda config --set show_channel_urls yesconda config --add channelshttps://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/ conda config --set show_channel_urls yes conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/ conda install pytorch torchvision cudatoolkit=10.1(这里是pytorch官网的命令去掉末尾的-c pytorch,去掉之后才是用清华源下载) |
今日新闻 |
推荐新闻 |
CopyRight 2018-2019 办公设备维修网 版权所有 豫ICP备15022753号-3 |